"""
@Time ： 2021/11/2 18:05
@Auth ： ChenZhangKun
"""
import random

import pygame

# 定义常量
# 定义屏幕大小
SCREEN_RECT = pygame.Rect(0, 0, 480, 700)
# 定义刷新的频率
FRAME_PER_SEC = 60
# 创建敌机的定时器常量
CREATE_ENEMY_EVENT = pygame.USEREVENT
# 英雄发射子弹时间
HERO_FIRE_EVENT = pygame.USEREVENT + 1


class GameSprite(pygame.sprite.Sprite):
    """
    飞机大战游戏精灵
    """

    def __init__(self, image_name, speed=1):
        # 调用父类的初始化方法
        super().__init__()
        # 定义属性
        self.image = pygame.image.load(image_name)
        self.rect = self.image.get_rect()
        self.speed = speed

    def update(self):
        # 垂直方向上移动
        self.rect.y += self.speed


class Background(GameSprite):
    """
    游戏背景精灵
    """

    def __init__(self, is_alt=False):
        # 调用父类方法
        super().__init__("./image/background.png")
        # 判断是否是交替图像，如果是设置初始位置
        if is_alt:
            self.rect.y = -self.rect.height

    def update(self):
        # 调用父类方法实现
        super().update()
        # 判断是否移除屏幕
        if self.rect.y >= SCREEN_RECT.height:
            self.rect.y = -self.rect.height


class Enemy(GameSprite):
    """
    敌机类
    """

    def __init__(self):
        # 调用父类方法，创建敌机精灵，指定敌机图片
        super().__init__("./image/enemy1.png")
        # z指定初始速度
        self.speed = random.randint(1, 3)
        # 初始位置
        self.rect.bottom = 0
        # 指定敌机的水平位置出现
        max_x = SCREEN_RECT.width - self.rect.width
        self.rect.x = random.randint(0, max_x)

    def update(self):
        super().update()
        if self.rect.y >= SCREEN_RECT.height:
            # 销毁敌机
            self.kill()

    def __del__(self):
        """
        敌机销毁时会调用
        :return:
        """
        pass


class Hero(GameSprite):
    """英雄精灵"""

    def __init__(self):
        # 调用父类方法调用图片和速度
        super().__init__("./image/me1.png", 0)
        # 设置英雄的初始位置
        self.rect.center = SCREEN_RECT.center
        # 底部位置
        self.rect.bottom = SCREEN_RECT.bottom - 120
        # 创建子弹的精灵组
        self.bullets = pygame.sprite.Group()

    def update(self):
        # 水平方向移动
        self.rect.x += self.speed
        # 控制左右边界
        if self.rect.x < 0:
            self.rect.x = 0
        elif self.rect.right > SCREEN_RECT.right:
            self.rect.right = SCREEN_RECT.right

    def fire(self):
        # 创建子弹
        for i in (0, 1, 2):
            bullet = Bullet()
            # 设置精灵位置
            bullet.rect.bottom = self.rect.y - i * 20
            bullet.rect.centerx = self.rect.centerx

            # 将精灵添加到精灵组
            self.bullets.add(bullet)


class Bullet(GameSprite):
    """子弹"""

    def __init__(self):
        super().__init__("./image/bullet1.png", -2)

    def update(self):
        # 向上飞行
        super().update()
        # 判断子弹是否飞出皮屏幕
        if self.rect.bottom < 0:
            self.kill()

    def __del__(self):
        print("子弹被销毁")
